package com.edu.base.common.utils;

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.Map;

import javax.imageio.ImageIO;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import com.ruoyi.common.utils.StringUtils;

public class LogoQRCodeUtils {
	
	private static final int IMAGE_WIDTH = 100;
	private static final int IMAGE_HEIGHT = 100;
	private static final int IMAGE_HALF_WIDTH = IMAGE_WIDTH / 2;
	private static final int FRAME_WIDTH = 2;
	private static MultiFormatWriter mutiWriter = new MultiFormatWriter();
    
   /**
	 * 生成二维码 并下载到本地
	 * @param content 内容
	 * @param width   宽
	 * @param height  高
	 * @param srcImagePath logo图片本地路径
	 * @param destImagePath 目的路径
	 * @param r 颜色rgb
	 * @param g 颜色rgb
	 * @param b 颜色rgb
	 * @param ifGradual 颜色是否渐变
	 */
	public static void encode(String content, int width, int height,
			String srcImagePath, String destImagePath, String r ,String g ,String b,boolean ifGradual) {
		   
		if(StringUtils.isBlank(r)&&StringUtils.isBlank(g)&&StringUtils.isBlank(b)){
		    	r="0";
		    	g="0";
		    	b="0";
		    }
		try {
			 int R=Integer.parseInt(r);
			 int G=Integer.parseInt(g);
			 int B=Integer.parseInt(b);
			ImageIO.write(genBarcode(content, width, height, srcImagePath ,R,G,B,ifGradual),
					"jpg", new File(destImagePath));
		} catch (IOException e) {
			e.printStackTrace();
		} catch (WriterException e) {
			e.printStackTrace();
		}
	}
    
	/**
	 * 生成二维码 并返回字节码数组，用于上传云
	 * @param content 内容
	 * @param width   宽
	 * @param height  高
	 * @param srcImagePath logo图片路径
	 * @param r  颜色rgb
	 * @param g  颜色rgb
	 * @param b  颜色rgb
	 * @param ifGradual 颜色是否渐变 
	 * @return
	 */
	public static byte[] encode(String content, int width, int height,
			String srcImagePath,String r ,String g ,String b ,boolean ifGradual) {
		byte[] byteArray=null;
		if(StringUtils.isBlank(r)&&StringUtils.isBlank(g)&&StringUtils.isBlank(b)){
	    	r="0";
	    	g="0";
	    	b="0";
	    }
		try {
			 int R=Integer.parseInt(r);
			 int G=Integer.parseInt(g);
			 int B=Integer.parseInt(b);
			ByteArrayOutputStream bsout = new ByteArrayOutputStream();
			ImageIO.write(genBarcode(content, width, height, srcImagePath ,R,G,B,ifGradual),
					"jpg", bsout);
			 byteArray = bsout.toByteArray();
			return byteArray;
		} catch (IOException e) {
			e.printStackTrace();
		} catch (WriterException e) {
			e.printStackTrace();
		}
		return byteArray;
	}

	
	private static BufferedImage genBarcode(String content, int width,
			int height, String srcImagePath,int r ,int g ,int b,boolean ifGradual) throws WriterException,
			IOException {
		BufferedImage scaleImage = scale(srcImagePath, IMAGE_WIDTH,
				IMAGE_HEIGHT, true);
		int[][] srcPixels = new int[IMAGE_WIDTH][IMAGE_HEIGHT];
		for (int i = 0; i < scaleImage.getWidth(); i++) {
			for (int j = 0; j < scaleImage.getHeight(); j++) {
				srcPixels[i][j] = scaleImage.getRGB(i, j);
			}
		}
		Map<EncodeHintType, Object> hint = new HashMap<EncodeHintType, Object>();
		hint.put(EncodeHintType.CHARACTER_SET, "utf-8");
		hint.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
		// 生成二维码
		BitMatrix matrix = mutiWriter.encode(content, BarcodeFormat.QR_CODE,
				width, height, hint);
		// 二维矩阵转为一维像素数组
		int halfW = matrix.getWidth() / 2;
		int halfH = matrix.getHeight() / 2;
		int[] pixels = new int[width * height];
		for (int y = 0; y < matrix.getHeight(); y++) {
			for (int x = 0; x < matrix.getWidth(); x++) {
				// 左上角颜色,根据自己需要调整颜色范围和颜色231, 144, 56
				if (x > 0 && x < 0 && y > 0 && y < 0) {
					Color color = new Color(r,g,b);
					int colorInt = color.getRGB();
					pixels[y * width + x] = matrix.get(x, y) ? colorInt
							: 16777215;
				}
				// 读取图片
				else if (x > halfW - IMAGE_HALF_WIDTH
						&& x < halfW + IMAGE_HALF_WIDTH
						&& y > halfH - IMAGE_HALF_WIDTH
						&& y < halfH + IMAGE_HALF_WIDTH) {
					pixels[y * width + x] = srcPixels[x - halfW
							+ IMAGE_HALF_WIDTH][y - halfH + IMAGE_HALF_WIDTH];
				} else if ((x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH
						&& x < halfW - IMAGE_HALF_WIDTH + FRAME_WIDTH
						&& y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH
						+ IMAGE_HALF_WIDTH + FRAME_WIDTH)
						|| (x > halfW + IMAGE_HALF_WIDTH - FRAME_WIDTH
								&& x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
								&& y > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH
								+ IMAGE_HALF_WIDTH + FRAME_WIDTH)
						|| (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH
								&& x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
								&& y > halfH - IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH
								- IMAGE_HALF_WIDTH + FRAME_WIDTH)
						|| (x > halfW - IMAGE_HALF_WIDTH - FRAME_WIDTH
								&& x < halfW + IMAGE_HALF_WIDTH + FRAME_WIDTH
								&& y > halfH + IMAGE_HALF_WIDTH - FRAME_WIDTH && y < halfH
								+ IMAGE_HALF_WIDTH + FRAME_WIDTH)) {
					pixels[y * width + x] = 0xfffffff;
					// 在图片四周形成边框
				} else {
					// 二维码颜色
					 Color color=null;
					 //是否颜色渐变
					if(ifGradual){
						int num1 = (int) (r - (r - r/2) / matrix.getHeight()* (y + 1));
						int num2 = (int) (g - (g - g/2) / matrix.getHeight()* (y + 1));
						int num3 = (int) (b - (b - b/2) / matrix.getHeight()* (y + 1));
					
					     color = new Color(num1, num2, num3);
					}else{
					     color = new Color(r,g,b);
					}
					int colorInt = color.getRGB();
					// 此处可以修改二维码的颜色，可以分别制定二维码和背景的颜色；
					pixels[y * width + x] = matrix.get(x, y) ? colorInt
							: 16777215;
					// 0x000000:0xffffff 16777215
				}
			}
		}
		BufferedImage image = new BufferedImage(width, height,
				BufferedImage.TYPE_INT_RGB);
		image.getRaster().setDataElements(0, 0, width, height, pixels);
		return image;
	}
    /**
     * 处理logo
     */
	private static BufferedImage scale(String srcImageFile, int height,
			int width, boolean hasFiller) throws IOException {
		double ratio = 0.0; // 缩放比例
		
		File file = new File(srcImageFile);
		BufferedImage srcImage = ImageIO.read(file);
		Image destImage = srcImage.getScaledInstance(width, height,
				BufferedImage.SCALE_SMOOTH);
		// 计算比例
		if ((srcImage.getHeight() > height) || (srcImage.getWidth() > width)) {
			if (srcImage.getHeight() > srcImage.getWidth()) {
				ratio = (new Integer(height)).doubleValue()
						/ srcImage.getHeight();
			} else {
				ratio = (new Integer(width)).doubleValue()
						/ srcImage.getWidth();
			}
			AffineTransformOp op = new AffineTransformOp(
					AffineTransform.getScaleInstance(ratio, ratio), null);
			destImage = op.filter(srcImage, null);
		}
		if (hasFiller) {
			// 补白
			BufferedImage image = new BufferedImage(width, height,
					BufferedImage.TYPE_INT_RGB);
			Graphics2D graphic = image.createGraphics();
			graphic.setColor(Color.white);
			graphic.fillRect(0, 0, width, height);
			if (width == destImage.getWidth(null))
				graphic.drawImage(destImage, 0,
						(height - destImage.getHeight(null)) / 2,
						destImage.getWidth(null), destImage.getHeight(null),
						Color.white, null);
			else
				graphic.drawImage(destImage,
						(width - destImage.getWidth(null)) / 2, 0,
						destImage.getWidth(null), destImage.getHeight(null),
						Color.white, null);
			graphic.dispose();
			destImage = image;
		}
		return (BufferedImage) destImage;
	}

	
	
	
	
	public static void main(String[] args) throws UnsupportedEncodingException {
		// 依次为内容(不支持中文),宽,长,中间图标路径,储存路径
		LogoQRCodeUtils.encode("http://www.baidu.com/", 512, 512,
				"D:\\logo.png", "D:\\2013-01.jpg","128","128","0",true);

	}
}